package com.ai.zuochengyun.phase01.class03;

/**
 * 找到后继节点
 */
public class Code16_FindSuccessorNode {

    private static class Node {
        int value;

        Node left;

        Node right;

        Node parent;
    }

    public static Node getSuccessorNode(Node node) {
        if (node == null) {
            return null;
        }

        // 有右树
        if (node.right != null) {
            return getLeftMost(node);
        }
        // 如果没有右树
        else {
            Node parent = node.parent;
            // parent.right == node 表示当前节点是其父亲节点的右孩子
            while (parent != null && parent.right == node) {
                // head来到parent的位置
                node = parent;
                // parent来到head.parent的位置
                parent = node.parent;
            }
            return parent;
        }
    }

    /**
     * 一直到最左
     * @param node
     * @return
     */
    private static Node getLeftMost(Node node) {
        if (node == null) {
            return null;
        }
        while (node.left != null) {
            node = node.left;
        }
        return node;
    }

}
